package compress

import (
	"bytes"
	"compress/gzip"
	"io/ioutil"

	"github.com/pkg/errors"
)

// GzipByteEncode 压缩
func GzipByteEncode(data []byte, levels ...int) ([]byte, error) {
	var b bytes.Buffer
	var level = gzip.DefaultCompression
	if len(levels) > 0 {
		level = levels[0]
	}

	w, err := gzip.NewWriterLevel(&b, level)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	if _, err := w.Write(data); err != nil {
		return nil, errors.WithStack(err)
	}

	if err := w.Flush(); err != nil {
		return nil, errors.WithStack(err)
	}

	if err := w.Close(); err != nil {
		return nil, errors.WithStack(err)
	}

	return b.Bytes(), nil
}

// GzipByteDecode 解压
func GzipByteDecode(data []byte) ([]byte, error) {
	r, err := gzip.NewReader(bytes.NewReader(data))
	if err != nil {
		return nil, errors.WithStack(err)
	}

	data, err = ioutil.ReadAll(r)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	if err := r.Close(); err != nil {
		return nil, errors.WithStack(err)
	}

	return data, nil
}
